Advent-Of-Code/2021/day17/day17 - part1.py at master · nalo26/Advent-Of-Code · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
file = open("input.txt")
x_area, y_area = file.readline()[13:].split(", ")
x_area = tuple(map(int, x_area[2:].split("..")))
y_area = tuple(map(int, y_area[2:].split("..")))


def vel_try(x, y):
    cur_pos = [0, 0]
    cur_vel = [x, y]
    ys = []
    while True:
        cur_pos[0] += cur_vel[0]
        cur_pos[1] += cur_vel[1]
        if cur_vel[0] > 0:
            cur_vel[0] -= 1
        cur_vel[1] -= 1
        ys.append(cur_pos[1])

        if (
            x_area[0] <= cur_pos[0] <= x_area[1]
            and y_area[0] <= cur_pos[1] <= y_area[1]
        ):
            return True, ys

        if cur_pos[0] > x_area[1] or cur_pos[1] < y_area[0]:
            return False, ys


y_dict = {}
for x in range(x_area[1] + 1):
    for y in range(1000):
        success, ys = vel_try(x, y)
        if success:
            y_dict[(x, y)] = max(ys)

print(max(y_dict.values()))